'''
    作者：刘妍
    版本：6.0
    日期：25/05/2020
    1.0功能：计算空气质量指数AQI
    2.0新增功能：读取已经获取的JSON数据文件，并将AQI前5的数据输出到文件
    3.0新增功能：读取已经获取的JSON数据文件，并将其转换为CSV文件
    4.0新增功能：自动识别文件类型（JSON文件还是CSV文件）
    5.0新增功能：利用网络爬虫实时获取城市的空气质量
    6.0新增功能：用BeautifulSoup模块更高效地访问和解析网页文件
'''

import requests
from bs4 import BeautifulSoup


# 自定义函数，获取城市的AQI
def get_city_aqi(city_pinyin):
    url = 'http://pm25.in/' + city_pinyin
    r = requests.get(url, timeout=30)
    # 创建BeautifulSoup对象
    soup = BeautifulSoup(r.text, 'lxml')
    # 按照属性查找所有满足条件的节点
    div_list = soup.find_all('div', {'class': 'span1'})

    city_aqi = []
    for i in range(8):
        div_content = div_list[i]
        caption = div_content.find('div', {'class': 'caption'}).text.strip()
        value = div_content.find('div', {'class': 'value'}).text.strip()
        # 将每一对caption和value作为一个元素增加到列表中
        city_aqi.append((caption, value))

    return city_aqi


def main():
    # 输入需要访问的城市拼音
    city_pinyin = input('请输入城市拼音：')
    # 调用自定义函数获取输入城市的AQI、PM25、PM10等值
    city_aqi = get_city_aqi(city_pinyin)
    # print(city_aqi)
    for i in city_aqi:
        print(i)


if __name__ == '__main__':
    main()